package com.cg.offer;

import com.cg.leetcode.ListNode;
import org.junit.Test;

/**
 * 剑指Offer 22.链表中倒数第k个节点
 *
 * @program: LeetCode->Offer_22
 * @description: 剑指Offer 22.链表中倒数第k个节点
 * @author: cg
 * @create: 2022-03-19 11:15
 **/
public class Offer_22 {

    @Test
    public void test22() {
        ListNode head = new ListNode(1);
        head.next = new ListNode(2);
        head.next.next = new ListNode(3);
        head.next.next.next = new ListNode(4);
        head.next.next.next.next = new ListNode(5);
        System.out.println(getKthFromEnd(head, 2));
    }

    /**
     * 输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。
     * <p>
     * 例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
     * <p>
     * 示例：
     * 给定一个链表: 1->2->3->4->5, 和 k = 2.
     * 返回链表 4->5.
     *
     * @param head
     * @param k
     * @return
     */
    public ListNode getKthFromEnd(ListNode head, int k) {
        ListNode latter = head, former = head;
        for (int i = 0; i < k; i++) {
            former = former.next;
        }
        while (former != null) {
            latter = latter.next;
            former = former.next;
        }
        return latter;
    }

}
